import os


def select_file(foldPath):
    global totalSize
    for fileName in os.listdir(foldPath):
        filePath = os.path.join(foldPath, fileName)
        if os.path.isfile(filePath):
            totalSize += os.path.getsize(filePath)
        if os.path.isdir(filePath):
            select_file(filePath)  # 若遇到文件夹则递归调用，以访问其子目录


while 1:
    foldPath = input('请输入需要查询的文件夹所在地址')
    if not foldPath:
        print('输入地址为空')

    if os.path.isfile(foldPath):
        print('地址错误，请勿传入具体文件地址')

    if os.path.isdir(foldPath):
        totalSize = 0
        select_file(foldPath)
        print('文件夹的大小为' + str(totalSize) + 'B')

    else:
        print('无效的地址')